package com.simple.gateway.pay.utils;
import org.apache.commons.codec.DecoderException;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.codec.binary.Hex;
import org.apache.commons.lang3.StringEscapeUtils;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;


/**
 * 封装各种格式的编码解码工具类.
 *
 * 1.Commons-Codec的 hex/base64 编码
 * 2.自制的base62 编码
 * 3.Commons-Lang的xml/html escape
 * 4.JDK提供的URLEncoder
 *
 * @author longww
 *
 */
public class Encodes {
    private static final String DEFAULT_URL_ENCODING = "UTF-8";
    private static final char[] BASE62 = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz".toCharArray();

    /**
     * 字节数字转Hex编码，输出转换后的Hex进制字符串.
     *
     * @param input 字节数字
     *
     * @return 输出转换后的Hex进制字符串
     */
    public static String encodeHex(byte[] input) {
        return Hex.encodeHexString(input);
    }

    /**
     * Hex解码.
     *
     * @param input 16进制字符串
     *
     * @return 输出转换后的字节数字
     */
    public static byte[] decodeHex(String input) {
        try {
            return Hex.decodeHex(input.toCharArray());
        } catch (DecoderException e) {
            throw ExceptionUtil.unchecked(e);
        }
    }



    /**
     * Base64编码.
     * @param input 字节数组
     * @return 返回Base64后的字符串
     */
    public static String encodeBase64(byte[] input) {
        String base64 = Base64.encodeBase64String(input);
        base64 = base64.replaceAll("\r", "");
        base64 = base64.replaceAll("\n", "");
        return base64;
    }

    /**
     * Base64编码, URL安全(将Base64中的URL非法字符'+'和'/'转为'-'和'_', 见RFC3548).
     * @param input 字节数组
     * @return 返回Base64后的字符串
     */
    public static String encodeUrlSafeBase64(byte[] input) {
        return Base64.encodeBase64URLSafeString(input);
    }

    /**
     * Base64解码.
     */
    public static byte[] decodeBase64(String input) {
        return Base64.decodeBase64(input);
    }

    /**
     * Base62编码。
     */
    public static String encodeBase62(byte[] input) {
        char[] chars = new char[input.length];
        for (int i = 0; i < input.length; i++) {
            chars[i] = BASE62[(input[i] & 0xFF) % BASE62.length];
        }
        return new String(chars);
    }

    /**
     * Html 转码.
     * @param html html字符串
     * @return 返回去除html字符的字符串
     */
    public static String escapeHtml(String html) {
        return StringEscapeUtils.escapeHtml4(html);
    }


    /**
     * Html 解码.
     * @param htmlEscaped 待解码的html字符串
     * @return 解码后的的字符串
     */
    public static String unescapeHtml(String htmlEscaped) {
        return StringEscapeUtils.unescapeHtml4(htmlEscaped);
    }


    /**
     * Xml 转码.
     * @param xml xml字符串
     * @return 转码后的字符串
     */
    public static String escapeXml(String xml) {
        return StringEscapeUtils.escapeXml(xml);
    }


    /**
     * Xml 转码.
     * @param xml 代码转码的xml字符串
     * @return 转码后的xml字符串
     */
    public static String escapeXml10(String xml) {
        return StringEscapeUtils.escapeXml10(xml);
    }

    /**
     * Xml 转码.
     * @param xml xml字符串
     * @return 转码后的字符串
     */
    public static String escapeXml11(String xml) {
        return StringEscapeUtils.escapeXml11(xml);
    }

    /**
     * Xml 解码.
     * @param xmlEscaped 待解码的xml字符串
     * @return 解码后的字符串
     */
    public static String unescapeXml(String xmlEscaped) {
        return StringEscapeUtils.unescapeXml(xmlEscaped);
    }


    /**
     * URL 编码, Encode默认为UTF-8.
     * @param part 代码编码的字符串（UTF-8编码）
     * @return 编码后的字符串
     */
    public static String urlEncode(String part) {
        try {
            return URLEncoder.encode(part, DEFAULT_URL_ENCODING);
        } catch (UnsupportedEncodingException e) {
            throw ExceptionUtil.unchecked(e);
        }
    }

    /**
     * URL 解码, Encode默认为UTF-8.
     * @param part 待解码的字符串（UTF-8编码）
     * @return 解码后的字符串
     */
    public static String urlDecode(String part) {

        try {
            return URLDecoder.decode(part, DEFAULT_URL_ENCODING);
        } catch (UnsupportedEncodingException e) {
            throw ExceptionUtil.unchecked(e);
        }
    }

    /**
     * 将字符串用MD5加密，并且将结果以16进制字符串形式输出
     *
     * @param data
     *            待加密的字符串
     * @return 以16进制字符串输出加密结果
     */
    public static final String encodeMD5(String data) {
        try {
            MessageDigest digest = MessageDigest.getInstance("MD5");
            digest.update(data.getBytes());
            return encodeHex(digest.digest());
        } catch (NoSuchAlgorithmException nsae) {
            System.err.println("Failed to load the MD5 MessageDigest. "
                    + "We will be unable to function normally.");
            nsae.printStackTrace();
            throw ExceptionUtil.unchecked(nsae);
        }

    }
}
